from WSOL.api.test import test
import yaml
import sys

if __name__ == '__main__':
    yaml_file = open('./configs/base_detector.yaml', 'r', encoding='utf-8')
    cfg = yaml.load(yaml_file)
    yaml_file.close()

    argv = sys.argv
    if len(argv) > 1:
        check_point_path = argv[1]
        cfg['Test_cfg']['check_point'] = check_point_path
    test(cfg)